#ifndef _COMM_HPP_
#define _COMM_HPP_

#include <iostream>
#include <cstdio>
#include <cerrno>
#include <cstring>
#include <sys/ipc.h>
#include <sys/shm.h>
#include <unistd.h>

#define PATHNAME "."
#define PROJ_ID 0x66
#define MAX_SIZE 4096

key_t getKey()
{
    key_t k = ftok(PATHNAME, PROJ_ID);
    if (k < 0)
    {
        std::cerr << errno << ":" << strerror(errno) << std::endl;
        exit(1);
    }
    return k;
}

int getShmHelper(key_t k, int flags)
{
    int shmid = shmget(k, MAX_SIZE, flags);
    if(shmid<0)
    {
        std::cerr << errno << ":" << strerror(errno) << std::endl;
        exit(2);
    }
    return shmid;
}

int creatShm(key_t k)
{
    return getShmHelper(k, IPC_CREAT | IPC_EXCL | 0600); // 创建一个新的，所以如果没有就判错
}

int getShm(key_t k)
{
    return getShmHelper(k, IPC_CREAT /*0*/); // 调用创建的,0也可以
}

void* attchShm(int shmid)
{
    void* mem = shmat(shmid,nullptr,0);
    if((long long)mem==-1L)
    {
        std::cerr << errno << ":" << strerror(errno) << std::endl;
        exit(3);
    }
    return mem;
}

void detachShm(void* start)
{
    if(shmdt(start)==-1)
    {
        std::cerr << errno << ":" << strerror(errno) << std::endl;
    }
}

void deleteShm(int shmid)
{
    if(shmctl(shmid,IPC_RMID,nullptr)==-1)
    {
        std::cerr << errno << ":" << strerror(errno) << std::endl;
        exit(1);
    }
}

#endif